from .dsgn2_backbone import DSGN2Backbone
from .dsgn2_backbone_transformer_fpn import DSGN2BackboneTransFPN
from .dsgn2_fast import DSGN2BackboneFast
from .dsgn2_fast_trt import DSGN2BackboneFastTRT
from .dsgn2_backbone_cnn import DSGN2BackboneCNN
from .dsgn2_backbone_cnn2d import DSGN2BackboneCNN2D, DSGN2BackboneCNN2DHalf
from .dsgn2_backbone_cnn_trt import DSGN2BackboneCNNTRT
from .dsgn2_backbone_cnn2d_trt import DSGN2BackboneCNN2DTRT, DSGN2BackboneCNN2DHalfTRT

__all__ = {
    'DSGN2Backbone': DSGN2Backbone,
    'DSGN2BackboneTransFPN': DSGN2BackboneTransFPN,
    'DSGN2BackboneFast': DSGN2BackboneFast,
    'DSGN2BackboneFastTRT':DSGN2BackboneFastTRT,
    'DSGN2BackboneCNN': DSGN2BackboneCNN,
    'DSGN2BackboneCNNTRT': DSGN2BackboneCNNTRT,
    'DSGN2BackboneCNN2D': DSGN2BackboneCNN2D,
    'DSGN2BackboneCNN2DHalf':DSGN2BackboneCNN2DHalf,
    'DSGN2BackboneCNN2DTRT': DSGN2BackboneCNN2DTRT,
    'DSGN2BackboneCNN2DHalfTRT':DSGN2BackboneCNN2DHalfTRT,
}